import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

# 创建一个示例数据矩阵
data = np.random.rand(10, 10)
df = pd.DataFrame(data, index=np.arange(10), columns=np.arange(10))
plt.figure(figsize=(10, 10))  # 设置图像大小
heatmap = plt.imshow(df, cmap='coolwarm')  # 使用colormap调整颜色映射
plt.colorbar(shrink=0.8)  # 添加颜色条
plt.title('Heatmap Example')  # 图表标题
plt.xlabel('Columns')
plt.ylabel('Rows')
plt.xticks(rotation=45)  # 倾斜x轴标签以适应长数值
plt.yticks(rotation=0)  # y轴标签保持水平
plt.show()
